__author__ = 'YinDu'

import random  # 随机数模块
import requests


def requests_headers():
    head_connection = ['Keep-Alive', 'close']
    head_accept = ['text/html,application/xhtml+xml,*/*']
    head_accept_language = ['zh-CN,fr-FR;q=0.5', 'en-US,en;q=0.8,zh-Hans-CN;q=0.5,zh-Hans;q=0.3']
    head_user_agent = [
        'Opera/9.27 (Windows NT 5.2; U; zh-cn)',
        'Opera/8.0 (Macintosh; PPC Mac OS X; U; en)',
        'Mozilla/5.0 (Macintosh; PPC Mac OS X; U; en) Opera 8.0',
        'Mozilla/4.0 (compatible; MSIE 8.0; Windows NT 6.1; Trident/4.0)',
        'Mozilla/5.0 (Windows NT 6.3; WOW64; Trident/7.0; rv:11.0) like Gecko',
        'Mozilla/5.0 (Windows; U; Windows NT 5.2) Gecko/2008070208 Firefox/3.0.1',
        'Mozilla/5.0 (Windows; U; Windows NT 5.1) Gecko/20070309 Firefox/2.0.0.3',
        'Mozilla/5.0 (Windows; U; Windows NT 5.1) Gecko/20070803 Firefox/1.5.0.12',
        'Mozilla/5.0 (Windows NT 6.1; WOW64; rv:21.0) Gecko/20100101 Firefox/21.0 ',
        'Mozilla/4.0 (compatible; MSIE 8.0; Windows NT 6.1; Win64; x64; Trident/4.0)',
        'Mozilla/5.0 (compatible; MSIE 10.0; Windows NT 6.1; WOW64; Trident/6.0; BIDUBrowser 2.x)',
        'Mozilla/5.0 (Windows NT 5.1) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/28.0.1500.95 Safari/537.36',
        'Mozilla/5.0 (Windows; U; Windows NT 5.1; en-US; rv:1.8.1.12) Gecko/20080219 Firefox/2.0.0.12 Navigator/9.0.0.6',
        'Mozilla/5.0 (Windows NT 6.1; WOW64) AppleWebKit/537.1 (KHTML, like Gecko) Chrome/21.0.1180.92 Safari/537.1 LBBROWSER',
        'Mozilla/5.0 (Windows NT 6.1; WOW64) AppleWebKit/536.11 (KHTML, like Gecko) Chrome/20.0.1132.11 TaoBrowser/3.0 Safari/536.11',
        'Mozilla/5.0 (Windows NT 6.1; WOW64) AppleWebKit/537.1 (KHTML, like Gecko) Maxthon/4.0.6.2000 Chrome/26.0.1410.43 Safari/537.1 ',
        'Mozilla/5.0 (Windows NT 6.1; WOW64; Trident/7.0; SLCC2; .NET CLR 2.0.50727; .NET CLR 3.5.30729; .NET CLR 3.0.30729; Media Center PC 6.0; .NET4.0C; rv:11.0) like Gecko)',
        'Mozilla/5.0 (compatible; MSIE 10.0; Windows NT 6.1; WOW64; Trident/6.0; SLCC2; .NET CLR 2.0.50727; .NET CLR 3.5.30729; .NET CLR 3.0.30729; Media Center PC 6.0; InfoPath.2; .NET4.0C; .NET4.0E)',
        'Mozilla/5.0 (compatible; MSIE 10.0; Windows NT 6.1; WOW64; Trident/6.0; SLCC2; .NET CLR 2.0.50727; .NET CLR 3.5.30729; .NET CLR 3.0.30729; Media Center PC 6.0; InfoPath.2; .NET4.0C; .NET4.0E; QQBrowser/7.3.9825.400)',
    ]

    # header 为随机产生一套由上边信息的header文件
    header = {
        'Accept': head_accept[0],
        'User-Agent': random.choices(head_user_agent)[0],
        'Connection': random.choices(head_connection)[0],
        'Accept-Language': random.choices(head_accept_language)[0],
    }
    return header  # 返回值为 header这个字典


def error_request(func):
    """
        The request processing
    """

    def inner(*args, **kwargs):
        try:
            return func(*args, **kwargs)
        except:
            return {'error': 20000, 'message': '网路出现问题'}

    return inner


class RequestWeb:
    def ret_text(self, ret):
        if ret.status_code == 200:
            try:
                text = ret.json()
            except:
                text = ret.text
        else:
            text = {'error': ret.status_code, 'message': '访问时出问题'}
        return text

    def request_tool(self, headers, proxies, rheaders, rproxies):
        if proxies:
            if rproxies:
                proxies = proxies
            else:
                proxies = requests_headers()
        else:
            proxies = requests_headers()
        if headers:
            if rheaders:
                headers = headers
            else:
                headers = requests_headers()
        else:
            headers = requests_headers()
        return headers, proxies

    @error_request
    def request_get(self, url, headers=None, proxies=None, rheaders=False, rproxies=False):
        headers, proxies = self.request_tool(headers, proxies, rheaders, rproxies)
        ret = requests.get(url=url, headers=headers, proxies=proxies, timeout=10)
        return self.ret_text(ret)

    @error_request
    def request_post(self, url, param, headers=None, proxies=None, rheaders=False, rproxies=False):
        headers, proxies = self.request_tool(headers, proxies, rheaders, rproxies)
        ret = requests.post(url, json=param, headers=headers, proxies=proxies, timeout=10)
        return self.ret_text(ret)


rweb = RequestWeb()
